[−][src]Crate tensorflow
This crate provides Rust bindings for the
TensorFlow
machine learning library.
If you aren't sure how to use something, please see the examples folder.
Modules
io | A module for writing TFRecords, Tensorflow's preferred on-disk data format. |
Structs
AttrMetadata | AttrMetadata describes the value of an attribute on an operation. |
BFloat16 | BFloat16 provides a Rust type for BFloat16. |
Device | Metadata about a device. |
FetchToken | An opaque token for retrieving an output from a computation. |
Function | Function is a grouping of operations with defined inputs and outputs. Once created and added to graphs, functions can be invoked by creating an operation whose operation type matches the function name. |
FunctionOptions | Options that can be passed during function creation. |
Graph | Represents a computation graph. Graphs may be shared between sessions. Graphs are thread-safe when used as directed. |
ImportGraphDefOptions |
|
ImportGraphDefResults | ImportGraphDefResults holds results that are generated by Graph::import_graph_def_with_results(). |
Input | A |
Library | Dynamically loaded plugins. The C API doesn't provide a way to unload libraries, so nothing happens when this goes out of scope. |
Operation | An |
OperationDescription | An |
OperationIter | Iterator over the operations in a |
Output | A |
QInt8 | Quantized type for i8. |
QInt16 | Quantized type for i16. |
QInt32 | Quantized type for i32. |
QUInt8 | Quantized type for u8. |
QUInt16 | Quantized type for u16. |
SavedModelBundle | Aggregation type for a saved model bundle. |
Session | Manages a single graph and execution. |
SessionOptions | Options that can be passed during session creation. |
SessionRunArgs | Manages the inputs and outputs for a single execution of a graph. |
Shape | A Shape is the shape of a tensor. A Shape may be an unknown rank, or it may have a known rank with each dimension being known or unknown. |
Status | Holds error information when communicating with back and forth with |
Tensor | Holds a multi-dimensional array of elements of a single data type. |
WhileBuilder | A WhileBuilder is used to build a while loop. |
Enums
AttrType | Describes the type of the value of an attribute on an operation. |
Code | Error values that can be returned. |
DataType | Type of a single tensor element. |
Traits
TensorType | A Rust type that maps to a |
Functions
get_all_registered_kernels | Returns a serialized KernelList protocol buffer containing KernelDefs for all registered kernels. |
get_registered_kernels_for_op | Returns a serialized KernelList protocol buffer containing KernelDefs for
all kernels registered for the operation named |
version | Returns a string describing version information of the
|
Type Definitions
OutputToken | Deprecated Deprecated alias for FetchToken. |
Result | Convenience type for |
StepWithGraph | Deprecated Deprecated alias for SessionRunArgs. |